package middleware.kafka.init_kafka;

import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.clients.consumer.ConsumerConfig;
import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.ConsumerRecords;
import org.apache.kafka.clients.consumer.KafkaConsumer;
import org.apache.kafka.common.PartitionInfo;
import org.apache.kafka.common.TopicPartition;
import org.apache.kafka.common.serialization.StringDeserializer;

import java.time.Duration;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;
import java.util.concurrent.atomic.AtomicBoolean;

/**
 * @Author suxiaoxiao
 * @Date 2019/6/27 22:07
 * @Version 1.0
 */
@Slf4j
public class KafkaConsumerTopicPartitionTest {

    public static final String brokerList = "47.104.178.113:9092";
    public static final String topic = "topic-demo";
    public static final String groupId = "group.demo";

    public static Properties initConfig() {
        Properties props = new Properties();
        /**
         * 消费者从 broker 端获取的消息格式都是字节数组（byte[]）类型，
         * 所以需要执行相应的反序列化操作才能还原成原有的对象格式。
         */
        props.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
        props.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, StringDeserializer.class.getName());
        /**
         * 用来指定连接 Kafka 集群所需的 broker 地址清单，
         * 具体内容形式为 host1:port1,host2:post，
         * 可以设置一个或多个地址，
         * 中间用逗号隔开，此参数的默认值为“”。
         */
        props.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, brokerList);
        /**
         * 消费者隶属的消费组的名称，默认值为“”。如果设置为空，则会报出异常
         * 一般而言，这个参数需要设置成具有一定的业务意义的名称。
         */
        props.put(ConsumerConfig.GROUP_ID_CONFIG, groupId);
        /**
         * 这个参数用来设定 KafkaConsumer 对应的客户端id，默认值也为“”。
         * 如果客户端不设置，则 KafkaConsumer 会自动生成一个非空字符串，
         * 内容形式如“consumer-1”、“consumer-2”，
         * 即字符串“consumer-”与数字的拼接。
         */
        props.put(ConsumerConfig.CLIENT_ID_CONFIG, "consumer.client.id.demo");
        return props;
    }

    public static void main(String[] args) {
        Properties props = initConfig();
        KafkaConsumer<String, String> consumer = new KafkaConsumer(props);
        // 消费者订阅主题 方法1 subscribe()
        consumer.subscribe(Arrays.asList(topic));
        /**
         * 消费者订阅主题 方法2 assign()
         * 这个方法只接受一个参数 partitions，用来指定需要订阅的分区集合。
         * TopicPartition
         * partition: 分区
         * topic: 主题
         */
        consumer.assign(Arrays.asList(new TopicPartition("topic-demo", 0)));

        /**
         * 消费者订阅主题 方法3 assign()
         */
        List<TopicPartition> partitions = new ArrayList<>();
        List<PartitionInfo> partitionInfos = consumer.partitionsFor(topic);
        if (partitionInfos != null) {
            for (PartitionInfo tpInfo : partitionInfos) {
                partitions.add(new TopicPartition(tpInfo.topic(), tpInfo.partition()));
            }
        }
        consumer.assign(partitions);

        /**
         * 取消订阅
         * 分别对应上面三个 订阅方法
         */
        consumer.unsubscribe();
        consumer.subscribe(new ArrayList<>());
        consumer.assign(new ArrayList<>());

        try {

        } catch (Exception e) {
            log.error("occur exception ", e);
        } finally {
            consumer.close();
        }

    }

}
